pass and return tuple to function python